package 中等.排序;

import java.util.Arrays;
import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;

/**
 * 有两只老鼠和 n 块不同类型的奶酪，每块奶酪都只能被其中一只老鼠吃掉。
 * 下标为 i 处的奶酪被吃掉的得分为：
 * 如果第一只老鼠吃掉，则得分为 reward1[i] 。
 * 如果第二只老鼠吃掉，则得分为 reward2[i] 。
 * 给你一个正整数数组 reward1 ，一个正整数数组 reward2 ，和一个非负整数 k 。
 * 请你返回第一只老鼠恰好吃掉 k 块奶酪的情况下，最大 得分为多少。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/mice-and-cheese
 */
public class 老鼠和奶酪_2611 {

    public static void main(String[] args) {

    }

    /**
     * 排序+贪心+哈希表
     * 1，按照第一个老鼠-第二个老鼠得分的差值降序排序，第一个
     * 老鼠优先选择前面 k 块奶酪
     * 2，老鼠二选择剩下的全部奶酪
     * 注意：老鼠一只能选择 reword1 中的奶酪，老鼠二选择 reward2
     */
    public int miceAndCheese(int[] reward1, int[] reward2, int k) {
        int ans = 0;
        Integer[] temp = new Integer[reward1.length];
        for (int i = 0; i < temp.length; i++) {
            temp[i] = i;
        }
        Arrays.sort(temp, (a, b) ->
                (reward1[b] - reward2[b]) - (reward1[a] - reward2[a])
        );
        Set<Integer> visited = new HashSet<>();
        for (int i = 0; i < k; i++) {
            ans += reward1[temp[i]];
            visited.add(temp[i]);
        }
        for (int i = 0; i < reward2.length; i++) {
            if (!visited.contains(i)) {
                ans += reward2[i];
            }
        }
        return ans;
    }

    /**
     * 贪心+优先队列
     * 1，第一只老鼠优先吃第一个老鼠-第二个老鼠得分较高的，第二只老鼠吃剩下全部的
     * 2，优先队列中只保存 k 个奶酪，如果出现 k + 1 个奶酪，那么堆顶取出来的奶酪
     * 一定是被第二只老鼠吃的，最后优先队列中剩下的 k 个奶酪是第一只老鼠吃的
     */
    public int miceAndCheese2(int[] reward1, int[] reward2, int k) {
        int maxScore = 0;
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>((a, b) -> a[1] - b[1]);
        for (int i = 0; i < reward1.length; i++) {
            priorityQueue.add(new int[]{i, reward1[i] - reward2[i]});
            if (priorityQueue.size() > k) {
                maxScore += reward2[priorityQueue.poll()[0]];
            }
        }
        while (!priorityQueue.isEmpty()) {
            maxScore += reward1[priorityQueue.poll()[0]];
        }
        return maxScore;
    }
}
